#!/bin/bash
# Quick test to verify the worker count fix

echo "Testing with corrected worker handling..."
echo ""

PYTHONPATH= venv-np126/bin/python main.py \
    --train_data BSDS \
    --choose_test_data 1 \
    --input_dir /home/dd/Public/2440_labeled/ \
    --input_val_dir /home/dd/Public/2440_labeled/ \
    --output_dir "checkpoints/2440gray" \
    --train_list train_pair.lst \
    --test_list val_pair.lst \
    --disable_val_size_alignment \
    --test_img_height 2080 \
    --test_img_width 2448 \
    --grayscale \
    --crop_img true \
    --workers 96 \
    --use_augmentation --aug_flip --aug_rotation --aug_brightness --aug_contrast --aug_noise \
    --epochs 1800 \
    --save_interval 5 \
    --batch_size 30 \
    --lr 1e-3 \
    --amp \
    --hy_data_dir "/home/dd/Public/dexined_labeled/train/" \
    --hy_repeat 40 \
    --resume-latest

echo ""
echo "Exit code: $?"
